/*
 * Copyright (c) 2021 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "local_net_utils.h"

#include <stdio.h>
#include <stdlib.h>
#ifdef L0_DEVICE
#include "lwip/ip4_addr.h"
#include "lwip/netif.h"
#include "lwip/netifapi.h"
#elif defined(L2_DEVICE)
#include <arpa/inet.h>
#include <fcntl.h>
#include <net/if.h>
#include <netinet/in.h>
#include <string.h>
#include <sys/ioctl.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#endif

//加密计算符号等化
#define SHA256_ROTL(a, b) (((a >> (32 - b)) & (0x7fffffff >> (31 - b))) | (a << b))
#define SHA256_SR(a, b) ((a >> b) & (0x7fffffff >> (b - 1)))

#define SHA256_Ch(x, y, z) ((x & y) ^ ((~x) & z))
#define SHA256_Maj(x, y, z) ((x & y) ^ (x & z) ^ (y & z))
#define SHA256_E0(x) (SHA256_ROTL(x, 30) ^ SHA256_ROTL(x, 19) ^ SHA256_ROTL(x, 10))
#define SHA256_E1(x) (SHA256_ROTL(x, 26) ^ SHA256_ROTL(x, 21) ^ SHA256_ROTL(x, 7))
#define SHA256_O0(x) (SHA256_ROTL(x, 25) ^ SHA256_ROTL(x, 14) ^ SHA256_SR(x, 3))
#define SHA256_O1(x) (SHA256_ROTL(x, 15) ^ SHA256_ROTL(x, 13) ^ SHA256_SR(x, 10))

/**
 * @brief 计算字符串SHA-256
 *
 * @param str 字符串指针
 * @param length 字符串长度
 * @param sha256 用于保存SHA-256的字符串指针
 * @return int 成功返回0
 */
int StrSha256(char *str, long length, char *sha256)
{
    char *pp, *ppend;
    long l, i, W[64], T1, T2, B, C, F, G, H0, H1, H2, H3, H4, H5, H6, H7;
    H0 = 0x6a09e667, H1 = 0xbb67ae85, H2 = 0x3c6ef372, H3 = 0xa54ff53a;
    H4 = 0x510e527f, H5 = 0x9b05688c, H6 = 0x1f83d9ab, H7 = 0x5be0cd19;
    long K[64] = {
        0x428a2f98, 0x71374491, 0xb5c0fbcf, 0xe9b5dba5, 0x3956c25b, 0x59f111f1, 0x923f82a4, 0xab1c5ed5,
        0xd807aa98, 0x12835b01, 0x243185be, 0x550c7dc3, 0x72be5d74, 0x80deb1fe, 0x9bdc06a7, 0xc19bf174,
        0xe49b69c1, 0xefbe4786, 0x0fc19dc6, 0x240ca1cc, 0x2de92c6f, 0x4a7484aa, 0x5cb0a9dc, 0x76f988da,
        0x983e5152, 0xa831c66d, 0xb00327c8, 0xbf597fc7, 0xc6e00bf3, 0xd5a79147, 0x06ca6351, 0x14292967,
        0x27b70a85, 0x2e1b2138, 0x4d2c6dfc, 0x53380d13, 0x650a7354, 0x766a0abb, 0x81c2c92e, 0x92722c85,
        0xa2bfe8a1, 0xa81a664b, 0xc24b8b70, 0xc76c51a3, 0xd192e819, 0xd6990624, 0xf40e3585, 0x106aa070,
        0x19a4c116, 0x1e376c08, 0x2748774c, 0x34b0bcb5, 0x391c0cb3, 0x4ed8aa4a, 0x5b9cca4f, 0x682e6ff3,
        0x748f82ee, 0x78a5636f, 0x84c87814, 0x8cc70208, 0x90befffa, 0xa4506ceb, 0xbef9a3f7, 0xc67178f2,
    };

    l = length + ((length % 64 >= 56) ? (128 - length % 64) : (64 - length % 64));
    if (!(pp = (char *)malloc((unsigned long)l)))
        return 0;
    for (i = 0; i < length; pp[i + 3 - 2 * (i % 4)] = str[i], i++)
        ;
    for (pp[i + 3 - 2 * (i % 4)] = 128, i++; i < l; pp[i + 3 - 2 * (i % 4)] = 0, i++)
        ;
    *((long *)(pp + l - 4)) = length << 3;
    *((long *)(pp + l - 8)) = length >> 29;

    for (ppend = pp + l; pp < ppend; pp += 64){
        for (i = 0; i < 16; W[i] = ((long*)pp)[i], i++);
        for (i = 16; i < 64; W[i] = (SHA256_O1(W[i - 2]) + W[i - 7] + SHA256_O0(W[i - 15]) + W[i - 16]), i++);
        long A, D, E, H;
        A = H0, B = H1, C = H2, D = H3, E = H4, F = H5, G = H6, H = H7;
        for (i = 0; i < 64; i++) {
            T1 = H + SHA256_E1(E) + SHA256_Ch(E, F, G) + K[i] + W[i];
            T2 = SHA256_E0(A) + SHA256_Maj(A, B, C);
            H = G, G = F, F = E, E = D + T1, D = C, C = B, B = A, A = T1 + T2;
        }
        H0 += A, H1 += B, H2 += C, H3 += D, H4 += E, H5 += F, H6 += G, H7 += H;
    }
    free(pp - l);
    sprintf(sha256, "%08lx%08lx%08lx%08lx%08lx%08lx%08lx%08lx", H0, H1, H2, H3, H4, H5, H6, H7);
    return 0;
}

int getLocalIp(const char *eth_name, char *local_ip_addr)
{
    int ret = -1;
#ifdef L0_DEVICE
    struct netif *lwipNetif = netifapi_netif_find_by_name(eth_name);
    uint32_t ip = lwipNetif->ip_addr->ip4_addr_t->addr;
    LOG_I("net_intf_status_change_cb **ip = %s\n", inet_ntoa(ip));
    memset(local_ip_addr, 0, IP_STR_LEN_MAX + 1);

#elif defined(L2_DEVICE)
    register int fd;
    struct ifreq ifr[32];
    struct ifconf ifc;

    if (local_ip_addr == NULL || eth_name == NULL) {
        return ret;
    }
    if ((fd = socket(AF_INET, SOCK_DGRAM, 0)) > 0) {
        ifc.ifc_len = sizeof ifr;
        ifc.ifc_buf = (caddr_t)ifr;
        if (!ioctl(fd, SIOCGIFCONF, (char *)&ifc)) //获取所有接口信息
        {
            register int intrface = ifc.ifc_len / sizeof(struct ifreq);
            while (intrface-- > 0) {
                // Get IP Address
                if (!(ioctl(fd, SIOCGIFADDR, (char *)&ifr[intrface]))) {
                    if (strcmp(eth_name, ifr[intrface].ifr_name) == 0) {
                        ret = 0;
                        sprintf(local_ip_addr, "%s",
                                inet_ntoa(((struct sockaddr_in *)(&ifr[intrface].ifr_addr))->sin_addr));
                        break;
                    }
                }
            }
        }
    }
    if (fd > 0) {
        close(fd);
    }
#endif
    return ret;
}

#ifdef L2_DEVICE
int logFd;
int initLogs(const char *logRoute)
{
    logFd = open(logRoute, O_CREAT | O_APPEND | O_RDWR);
    if (logFd < 0) {
        return -1;
    }
    return logFd;
}

int addLogs(const char *logs)
{
    if (logFd >= 0 && logs) {
        write(logFd, logs, strlen(logs));
        write(logFd, "\n", strlen("\n"));
        fsync(logFd);
        return 0;
    }
    return -1;
}

u32 getMillionSecond(void)
{
    struct timeval tv;
    gettimeofday(&tv, NULL);

    return (tv.tv_sec * 1000) + (tv.tv_usec / 1000);
}

int closeLogs()
{
    if (logFd >= 0) {
        close(logFd);
        logFd = -1;
    }
    return 0;
}
#endif